Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Tensorflow slim Resnet101_v1 support #3

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

insikk
Copy link

@insikk insikk commented May 25, 2018

I added additional notebook for demo of tf-resnet support.
Please review the result, and merge into this repo.

Tensorflow support will definitely help deep learning community.

Thank you for sharing your base code.

@tchugh
Copy link

tchugh commented Aug 23, 2018

Hi Insikk,

I am trying to add support for Inception-v3 through tf-slim. Could you please tell me where can I get the layer mappings similar to "resnet_v1_101_caffe_to_tf" for Inception-v3?

Thanks,
Tarang

@insikk
Copy link
Author

insikk commented Aug 24, 2018

@tchugh
The resnet_v1_101 weight mapping is available because the resnet v1 tensorflow implementation is borrowing caffe's weight by design. What I did was comparing two model architecture of resnet in both tensorflow and caffe, then mapping layer by layer. Of course, there was minor modification that was handled by some tinkering.

However, inception networks are mainly built on tensorflow directly. So I don't think there is an easy way. You might able to find a caffe implementation of inception-v3 which uses its weight trained on tensorflow. Then, you can follow what I did for resnet cnn fixation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants